from sqlalchemy import Column, String

from api.utils.base import Base


class UserLLM(Base):
    __tablename__ = "t_user_llm"
    id: str = Column(String, primary_key=True)
    chat_model: str = Column(String, comment="聊天模型")
    embedding_model: str = Column(String, comment="embedding模型")
    img2txt_model: str = Column(String, comment="img2txt模型")
    sequence2txt_model: str = Column(String, comment="sequence2txt模型")
    rerank_model : str = Column(String, comment="rerank模型")
    tts_model : str = Column(String, comment="tts模型")
    user_id: str = Column(String, comment="用户id")


